package com.fire.config;

import java.util.ArrayList;
import java.util.List;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import springfox.documentation.builders.ApiInfoBuilder;
import springfox.documentation.builders.ParameterBuilder;
import springfox.documentation.builders.PathSelectors;
import springfox.documentation.builders.RequestHandlerSelectors;
import springfox.documentation.schema.ModelRef;
import springfox.documentation.service.ApiInfo;
import springfox.documentation.service.Parameter;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.spring.web.plugins.Docket;

@Configuration
public class Swagger2UI {
	@Bean
	public Docket createRestApi() {
		// 令牌
		ParameterBuilder token = new ParameterBuilder();
		List<Parameter> pars = new ArrayList<Parameter>();
		token.name("Token").description("请求令牌").modelRef(new ModelRef("string")).parameterType("header").required(false)
				.build(); // header中的token参数非必填，传空也可以
		pars.add(token.build()); // 根据每个方法名也知道当前方法在设置什么参数

		return new Docket(DocumentationType.SWAGGER_2).apiInfo(apiInfo()).select()
				.apis(RequestHandlerSelectors.basePackage("com.fire.controller")).paths(PathSelectors.any()).build()
				.globalOperationParameters(pars);
	}

	private ApiInfo apiInfo() {
		return new ApiInfoBuilder().title("造造团队-API文档").description("简单优雅的RESTful风格").version("1.0").build();
	}
}
